package com.cqhilink.api.common.utils.mail;

import com.sun.mail.util.MailSSLSocketFactory;

import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.mail.*;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeBodyPart;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import java.io.File;
import java.security.GeneralSecurityException;
import java.util.Date;
import java.util.Enumeration;
import java.util.Properties;
import java.util.Vector;

/**
 * 邮件发送器
 *
 * @author 王镇
 * @create 2017-08-03 17:18
 */
public class MailSender {

    /**
     * 以文本格式发送邮件
     *
     * @param mailInfo 待发送的邮件的信息
     * @return
     */
    public boolean sendTextMail(MailSenderInfo mailInfo) {
        // 判断是否需要身份认证
        MyAuthenticator authenticator = null;
        // 获得邮件会话属性
        Properties pro = mailInfo.getProperties();

        if (mailInfo.isValidate()) {
            // 如果需要身份认证，则创建一个密码验证器
            authenticator = new MyAuthenticator(mailInfo.getUserName(), mailInfo.getPassword());
        }
        // 根据邮件会话属性和密码验证器构造一个发送邮件的session
        Session sendMailSession = Session.getDefaultInstance(pro, authenticator);
        try {
            // 开启SSL加密
            MailSSLSocketFactory sf = new MailSSLSocketFactory();
            sf.setTrustAllHosts(true);
            pro.put("mail.smtp.ssl.enable", "true");
            pro.put("mail.smtp.ssl.socketFactory", sf);

            // 根据session创建一个邮件消息
            Message mailMessage = new MimeMessage(sendMailSession);
            // 创建邮件发送者地址    authenticator
            Address from = new InternetAddress(mailInfo.getFromAddress());
            // 设置邮件消息的发送者
            mailMessage.setFrom(from);
            // 创建邮件的接收者地址，并设置到邮件消息中
            String[] mailToAddress = mailInfo.getToAddress();
            int len = mailToAddress.length;
            Address to[] = new InternetAddress[len];
            for (int i = 0; i < len; i++) {
                to[i] = new InternetAddress(mailToAddress[i]);
            }
            // Message.RecipientType.TO属性表示接收者的类型为TO
            //TO表示主要接收人，CC表示抄送人，BCC表示秘密抄送人。
            mailMessage.setRecipients(Message.RecipientType.TO, to);
            // 设置邮件消息的主题
            mailMessage.setSubject(mailInfo.getSubject());
            // 设置邮件消息发送的时间
            mailMessage.setSentDate(new Date());
            // 设置邮件消息的主要内容
            String mailContent = mailInfo.getContent();
            mailMessage.setText(mailContent);
            // 发送邮件
            Transport.send(mailMessage);
            return true;
        } catch (MessagingException ex) {
            ex.printStackTrace();
        } catch (GeneralSecurityException e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 以HTML格式发送邮件
     *
     * @param mailInfo 待发送的邮件信息
     * @return
     */
    public boolean sendHtmlMail(MailSenderInfo mailInfo){
        // 判断是否需要身份认证
        MyAuthenticator authenticator = null;
        Properties pro = mailInfo.getProperties();


        //如果需要身份认证，则创建一个密码验证器
        if (mailInfo.isValidate()) {
            authenticator = new MyAuthenticator(mailInfo.getUserName(), mailInfo.getPassword());
        }
        // 根据邮件会话属性和密码验证器构造一个发送邮件的session
        Session sendMailSession = Session.getDefaultInstance(pro, authenticator);
        try {
            // 开启SSL加密
            MailSSLSocketFactory sf = new MailSSLSocketFactory();
            sf.setTrustAllHosts(true);
            pro.put("mail.smtp.ssl.enable", "true");
            pro.put("mail.smtp.ssl.socketFactory", sf);

            // 根据session创建一个邮件消息
            Message mailMessage = new MimeMessage(sendMailSession);
            // 创建邮件发送者地址
            Address from = new InternetAddress(mailInfo.getFromAddress());
            // 设置邮件消息的发送者
            mailMessage.setFrom(from);
            // 创建邮件的接收者地址，并设置到邮件消息中
            String[] mailToAddress = mailInfo.getToAddress();
            int len = mailToAddress.length;
            Address to[] = new InternetAddress[len];
            for (int i = 0; i < len; i++) {
                to[i] = new InternetAddress(mailToAddress[i]);
            }
            // Message.RecipientType.TO属性表示接收者的类型为TO
            //mailMessage.setRecipient(Message.RecipientType.TO, to) ;
            mailMessage.setRecipients(Message.RecipientType.TO, to);
            // 设置邮件消息的主题
            mailMessage.setSubject(mailInfo.getSubject());
            // 设置邮件消息发送的时间
            mailMessage.setSentDate(new Date());
            // MiniMultipart类是一个容器类，包含MimeBodyPart类型的对象
            Multipart mainPart = new MimeMultipart();
            // 创建一个包含HTML内容的MimeBodyPart
            BodyPart html = new MimeBodyPart();
            // 设置HTML内容
            html.setContent(mailInfo.getContent(), "text/html; charset=utf-8");
            mainPart.addBodyPart(html);
            //向multipart中添加附件
            Vector<File> file = mailInfo.getFile();
            String fileName = mailInfo.getFileName();
            Enumeration<File> efile = file.elements();
            while (efile.hasMoreElements()) {
                MimeBodyPart mdpFile = new MimeBodyPart();
                fileName = efile.nextElement().toString();
                FileDataSource fds = new FileDataSource(fileName);
                mdpFile.setDataHandler(new DataHandler(fds));
                //这个方法可以解决乱码问题
                String fileName1 = new String(fds.getName().getBytes(), "ISO-8859-1");
                mdpFile.setFileName(fileName1);
                mainPart.addBodyPart(mdpFile);
            }
            file.removeAllElements();
            // 将MiniMultipart对象设置为邮件内容
            mailMessage.setContent(mainPart);
            // 发送邮件
            Transport.send(mailMessage);
            return true;
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return false;
    }

    /**
     * main函数测试
     * @param args
     */
    public static void main(String[] args) {
        try {
            MailSenderInfo mailInfo = new MailSenderInfo();
            // 设置邮件服务器类型
//            mailInfo.setMailServerHost("smtp.qq.com");
//            mailInfo.setMailServerHost("smtp.exmail.qq.com");
            mailInfo.setMailServerHost("smtp.sina.com");

            //设置端口号
            mailInfo.setMailServerPort("465");
            //设置是否验证
            mailInfo.setValidate(true);
            //设置用户名、密码、发送人地址
//            mailInfo.setUserName("376919915@qq.com");
//            mailInfo.setPassword("kbkoxahgcmribjjg");// 您的邮箱密码
//            mailInfo.setFromAddress("376919915@qq.com");

//            mailInfo.setUserName("wangzhen@kingyun.net");
//            mailInfo.setPassword("Wz123456789");// 您的邮箱密码
//            mailInfo.setFromAddress("wangzhen@kingyun.net");

            mailInfo.setUserName("wangzhen_cq@sina.com");
            mailInfo.setPassword("cqwz123456");// 您的邮箱密码
            mailInfo.setFromAddress("wangzhen_cq@sina.com");
            /**
             * 附件
             */
//            Vector<File> file = new Vector<File>() ;
//            File f1 = new File("d:/1.txt") ;
//            File f2 = new File("d:/2.txt") ;
//            File f3 = new File("d:/cn.csv") ;
//            File f4 = new File("C:/Documents and Settings/Administrator/My Documents/Downloads/mail.jar/mail.jar") ;
//            file.add(f1) ;
//            file.add(f2) ;
//            file.add(f3) ;
//            file.add(f4) ;
//            mailInfo.setFile(file) ;
            /**
             * 收件人邮箱
             */
            String[] mailToAddress = {"349801753@qq.com","376919915@qq.com"} ;
            mailInfo.setToAddress(mailToAddress);
            mailInfo.setSubject("测试");
            mailInfo.setContent("注册地址 <a href=\"www.baidu.com\" target='_blank'>百度</a>");
            // 这个类主要来发送邮件
            MailSender sms = new MailSender();
            // sms.sendTextMail(mailInfo);// 发送文体格式
            if(sms.sendHtmlMail(mailInfo)){
                System.out.println("mail send True!");
            }// 发送html格式
            else{
                System.out.println("mail send False!");
            }
        } catch (Exception e) {
            e.printStackTrace() ;
        }
    }

}
